package commons;


import org.apache.commons.io.FileUtils;
import org.openqa.selenium.OutputType;
import org.openqa.selenium.TakesScreenshot;
import org.openqa.selenium.WebDriver;

import java.io.File;
import java.io.IOException;
import java.text.SimpleDateFormat;

/**
 * Created with IntelliJ IDEA.
 * Description :
 * User: jiume
 * Date: 2025-09-09
 * Time: 19:48
 */
// 保存通用的工具
public class Utils {

    /**
     * 获取屏幕截图
     * ./src/test/java/images/
     *                        年-月-日
     *                                /方法-时-分-秒-毫秒
     */
    public static void getScreenShot(WebDriver driver, String method) throws IOException {
        // 定义时间格式
        SimpleDateFormat s1 = new SimpleDateFormat("yy-MM-dd");
        SimpleDateFormat s2 = new SimpleDateFormat("HH-mm-ss-SS");
        // 获取对应的时间
        String path1 = s1.format(System.currentTimeMillis());
        String path2 = s2.format(System.currentTimeMillis());
        // 拼接对应的文件格式
        String path = "./src/test/java/images/" + path1 + "/" + method + "-" + path2 + ".png";
        // 截图
        File file = ((TakesScreenshot) driver).getScreenshotAs(OutputType.FILE);
        FileUtils.copyFile(file, new File(path));
    }
}
